package cool.tsy.emos.wx.service.impl;

import cn.hutool.core.date.DateField;
import cn.hutool.core.date.DateRange;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.http.HttpUtil;
import cool.tsy.emos.wx.constant.SystemConstants;
import cool.tsy.emos.wx.db.dao.*;
import cool.tsy.emos.wx.db.pojo.TbCheckin;
import cool.tsy.emos.wx.db.pojo.TbFaceModel;
import cool.tsy.emos.wx.exception.EmosException;
import cool.tsy.emos.wx.service.CheckinService;
import cool.tsy.emos.wx.task.EmailTask;
import lombok.extern.slf4j.Slf4j;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Scope;
import org.springframework.mail.SimpleMailMessage;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;

/**
 * <p>
 * 签到表 服务实现类
 * </p>
 *
 * @author WangHongxuan
 * @since 2021-09-24
 */
@Slf4j
@Service
@Scope("prototype")
@SuppressWarnings("all")
public class CheckinServiceImpl implements CheckinService {

    private final SystemConstants systemConstants;

    private final TbHolidaysDao holidaysDao;

    private final TbWorkdayDao workdayDao;

    private final TbCheckinDao checkinDao;

    private final TbFaceModelDao faceModelDao;

    private final TbCityDao cityDao;

    private final TbUserDao userDao;

    private final EmailTask emailTask;

    @Value("${emos.face.createFaceModelUrl}")
    private String createFaceModelUrl;

    @Value("${emos.face.checkinUrl}")
    private String checkinUrl;

    @Value("${emos.email.hr}")
    private String hrEmail;

    @Value("${emos.code}")
    private String code;

    public CheckinServiceImpl(SystemConstants systemConstants, TbHolidaysDao holidaysDao, TbWorkdayDao workdayDao, TbCheckinDao checkinDao, TbFaceModelDao faceModelDao, TbCityDao cityDao, EmailTask emailTask, TbUserDao userDao) {
        this.systemConstants = systemConstants;
        this.holidaysDao = holidaysDao;
        this.workdayDao = workdayDao;
        this.checkinDao = checkinDao;
        this.faceModelDao = faceModelDao;
        this.cityDao = cityDao;
        this.emailTask = emailTask;
        this.userDao = userDao;
    }

    @Override
    public String validCanCheckIn(int userId, String date) {
        boolean bool_1 = holidaysDao.searchTodayIsHolidays() != null;
        boolean bool_2 = workdayDao.searchTodayIsWorkday() != null;
        String type = "工作日";
        if (DateUtil.date().isWeekend()) type = "节假日";
        if (bool_1) type = "节假日";
        else if(bool_2) type = "工作日";
        if (type.equals("节假日")) return "节假日不需要考勤";
        else {
            DateTime now = DateUtil.date();
            String start = DateUtil.today() + " " + systemConstants.attendanceStartTime;
            String end = DateUtil.today() + " " + systemConstants.attendanceEndTime;
            DateTime attendanceStart = DateUtil.parse(start);
            DateTime attendanceEnd = DateUtil.parse(end);
            if (now.isBefore(attendanceStart)) return "没到上班考勤开始时间";
            else if (now.isAfter(attendanceEnd)) return "超过了上班考勤结束时间";
            else {
                HashMap<String, Object> param = new HashMap<>();
                param.put("userId", userId);
                param.put("date", date);
                param.put("start", start);
                param.put("end", end);
                boolean existCheckIn = checkinDao.haveCheckin(param) != null;
                return existCheckIn ? "今日已经考勤，不用重复考勤" : "可以考勤";
            }
        }
    }

    @Override
    public void checkin(HashMap param) {
        // 当前时间
        DateTime curTime = DateUtil.date();
        // 上班时间
        DateTime attendanceTime = DateUtil.parse(DateUtil.today() + " " + systemConstants.attendanceTime);
        // 上班考勤结束时间
        DateTime attendanceEndTime = DateUtil.parse(DateUtil.today() + " " + systemConstants.attendanceEndTime);
        // 保存考勤结果
        int status = 1;
        if (curTime.compareTo(attendanceTime) <= 0) {
            // 正常考勤
            status = 1;
        } else if (curTime.compareTo(attendanceTime) > 0 && curTime.compareTo(attendanceEndTime) < 0) {
            // 迟到
            status = 2;
        }
        int userId = (Integer) param.get("userId");
        String faceModel = faceModelDao.searchFaceModel(userId);
        if (StrUtil.isEmpty(faceModel)) {
            throw new EmosException("不存在人脸模型");
        } else {
            String path = (String) param.get("path");
            HttpRequest request = HttpUtil.createPost(checkinUrl);
            request.form("photo", FileUtil.file(path), "targetModel", faceModel);
            request.form("code", code);
            HttpResponse response = request.execute();
            if (response.getStatus() != 200) {
                log.error("人脸识别服务异常");
                throw new EmosException("人脸识别服务异常");
            }
            String body = response.body();
            if ("无法识别出人脸".equals(body) || "照片中存在多张人脸".equals(body)) {
                throw new EmosException(body);
            } else if ("False".equals(body)) {
                throw new EmosException("签到无效，非本人签到");
            } else if ("True".equals(body)) {
                // 查询疫情风险等级
                // 风险等级记录
                int risk = 1;
                String city = (String) param.get("city");
                String district = (String) param.get("district");
                String address = (String) param.get("address");
                String country = (String) param.get("country");
                String province = (String) param.get("province");
                if (!StrUtil.isBlank(city) && !StrUtil.isBlank(district)) {
                    String code = cityDao.searchCode(city);
                    try {
                        String url = "http://m." + code + ".bendibao.com/news/yqdengji/?qu=" + district;
                        Document document = Jsoup.connect(url).get();
                        Elements elements = document.getElementsByClass("list-content");
                        if (elements.size() > 0) {
                            Element element = elements.get(0);
                            String res = element.select("p:last-child").text();
                            if ("高风险".equals(res)) {
                                risk = 3;
                                // 发送告警邮件
                                HashMap<String, String> map = userDao.searchNameAndDept(userId);
                                String name = map.get("name");
                                String deptName = map.get("dept_name");
                                deptName = deptName != null ? deptName : "";
                                SimpleMailMessage message = new SimpleMailMessage();
                                message.setTo(hrEmail);
                                message.setSubject("员工" + name + "身处高风险疫情地区警告");
                                message.setText(deptName + "员工" + name + "，" + DateUtil.format(new Date(), "yyyy年MM月dd日") + "处于" + address + "，属于新冠疫情高风险地区，请及时与该员工联系，核实情况！");
                                emailTask.sendAsync(message);
                            } else if ("中风险".equals(res)) {
                                risk = 2;
                            }
                        }
                    } catch (Exception e) {
                        log.error("执行异常", e);
                        throw new EmosException("获取风险等级失败");
                    }
                }
                // 保存签到记录
                TbCheckin tbCheckin = new TbCheckin();
                tbCheckin.setUserId(userId);
                tbCheckin.setAddress(address);
                tbCheckin.setCountry(country);
                tbCheckin.setProvince(province);
                tbCheckin.setCity(city);
                tbCheckin.setDistrict(district);
                tbCheckin.setStatus((byte) status);
                tbCheckin.setRisk(risk);
                tbCheckin.setDate(DateUtil.today());
                tbCheckin.setCreateTime(curTime);
                checkinDao.insert(tbCheckin);
            }
        }
    }

    @Override
    public void createFaceModel(int userId, String path) {
        HttpRequest request = HttpUtil.createPost(createFaceModelUrl);
        request.form("photo", FileUtil.file(path));
        request.form("code", code);
        HttpResponse response = request.execute();
        String body = response.body();
        if ("无法识别出人脸".equals(body) || "照片中存在多张人脸".equals(body)) {
            throw new EmosException(body);
        } else {
            TbFaceModel faceModel = new TbFaceModel();
            faceModel.setUserId(userId);
            faceModel.setFaceModel(body);
            faceModelDao.insert(faceModel);
        }
    }

    @Override
    public HashMap searchTodayCheckin(int userId) {
        HashMap map = checkinDao.searchTodayCheckin(userId);
        return map;
    }

    @Override
    public long searchCheckinDays(int userId) {
        long days = checkinDao.searchCheckinDays(userId);
        return days;
    }

    @Override
    public ArrayList<HashMap> searchWeekCheckin(HashMap param) {
        ArrayList<HashMap> checkinList = checkinDao.searchWeekCheckin(param);
        ArrayList<String> holidaysList = holidaysDao.searchHolidaysInRange(param);
        ArrayList<String> workdayList = workdayDao.searchWorkdayInRange(param);
        DateTime startDate = DateUtil.parseDate(param.get("startDate").toString());
        DateTime endDate = DateUtil.parseDate(param.get("endDate").toString());
        DateRange range = DateUtil.range(startDate, endDate, DateField.DAY_OF_MONTH);
        ArrayList<HashMap> list = new ArrayList();
        range.forEach(item -> {
            String date = item.toString("yyyy-MM-dd");
            String type = "工作日";
            if (item.isWeekend()) {
                type = "节假日";
            }
            if (holidaysList != null && holidaysList.contains(date)) {
                type = "节假日";
            } else if (workdayList != null && workdayList.contains(date)) {
                type = "工作日";
            }
            String status = "";
            if ("工作日".equals(type) && DateUtil.compare(item, DateUtil.date()) <= 0) {
                status = "缺勤";
                boolean flag = false;
                for (HashMap<String, String> map : checkinList) {
                    if (map.containsValue(date)) {
                        status = map.get("status");
                        flag = true;
                        break;
                    }
                }
                DateTime endTime = DateUtil.parse(DateUtil.today() + " " + systemConstants.attendanceEndTime);
                String today = DateUtil.today();
                if (date.equals(today) && DateUtil.date().isBefore(endTime) && !flag) {
                    status = "";
                }
            }
            HashMap map = new HashMap();
            map.put("date", date);
            map.put("status", status);
            map.put("type", type);
            map.put("day", item.dayOfWeekEnum().toChinese("周"));
            list.add(map);
        });
        return list;
    }

    @Override
    public ArrayList<HashMap> searchMonthCheckin(HashMap param) {
        return searchWeekCheckin(param);
    }
}
